//--------------------------------------------------------------------------
//Copyright (c) 1998-2004, Drew Davidson and Luke Blanshard
//All rights reserved.

//Redistribution and use in source and binary forms, with or without
//modification, are permitted provided that the following conditions are
//met:

//Redistributions of source code must retain the above copyright notice,
//this list of conditions and the following disclaimer.
//Redistributions in binary form must reproduce the above copyright
//notice, this list of conditions and the following disclaimer in the
//documentation and/or other materials provided with the distribution.
//Neither the name of the Drew Davidson nor the names of its contributors
//may be used to endorse or promote products derived from this software
//without specific prior written permission.

//THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
//"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
//LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
//FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
//COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
//INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
//BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
//OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
//AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
//OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
//THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
//DAMAGE.
//--------------------------------------------------------------------------
package ognl;

import ognl.enhance.ExpressionCompiler;
import ognl.enhance.OrderedReturn;
import ognl.enhance.UnsupportedCompilationException;

import java.lang.reflect.Method;

/**
 * @author Luke Blanshard (blanshlu@netscape.net)
 * @author Drew Davidson (drew@ognl.org)
 */
public class ASTMethod extends SimpleNode implements OrderedReturn, NodeType
{

    private String _methodName;

    private String _lastExpression;

    private String _coreExpression;

    private Class _getterClass;

    public ASTMethod(int id)
    {
        super(id);
    }

    public ASTMethod(OgnlParser p, int id)
    {
        super(p, id);
    }

    /** Called from parser action. */
    public void setMethodName(String methodName)
    {
        _methodName = methodName;
    }

    /**
     * Returns the method name that this node will call.
     */
    public String getMethodName()
    {
        return _methodName;
    }

    protected Object getValueBody(OgnlContext context, Object source)
    throws OgnlException
    {
        Object[] args = OgnlRuntime.getObjectArrayPool().create(jjtGetNumChildren());

        try {
            Object result, root = context.getRoot();

            for(int i = 0, icount = args.length; i < icount; ++i) {
                args[i] = _children[i].getValue(context, root);
            }
            
            result = OgnlRuntime.callMethod(context, source, _methodName, null, args);
            
            if (result == null)
            {
                NullHandler nh = OgnlRuntime.getNullHandler(OgnlRuntime.getTargetClass(source));
                result = nh.nullMethodResult(context, source, _methodName, args);
            }

            return result;
            
        } finally {
            OgnlRuntime.getObjectArrayPool().recycle(args);
        }
    }

    public String getLastExpression()
    {
        return _lastExpression;
    }

    public String getCoreExpression()
    {
        return _coreExpression;
    }

    public Class getGetterClass()
    {
        return _getterClass;
    }

    public Class getSetterClass()
    {
        return _getterClass;
    }

    public String toString()
    {
        String result = _methodName;

        result = result + "(";
        if ((_children != null) && (_children.length > 0)) {

            for(int i = 0; i < _children.length; i++) {
                if (i > 0) {
                    result = result + ", ";
                }

                result = result + _children[i];
            }
        }

        result = result + ")";
        return result;
    }

    public String toGetSourceString(OgnlContext context, Object target)
    {
      /*  System.out.println("methodName is " + _methodName + " for target " + target + " target class: " + (target != null ? target.getClass() : null)
                           + " current type: " + context.getCurrentType());*/
        if (target == null)
            throw new UnsupportedCompilationException("Target object is null.");

        String post = "";
        String result = null;
        Method m = null;
        
        try {
            
            m = OgnlRuntime.getMethod(context, context.getCurrentType() != null ? context.getCurrentType() : target.getClass(), _methodName, _children, false);
            if (m == null)
                m = OgnlRuntime.getReadMethod(target.getClass(), _methodName, _children != null ? _children.length : -1);
            
            if (m == null) {
                m = OgnlRuntime.getWriteMethod(target.getClass(), _methodName, _children != null ? _children.length : -1);

                if (m != null) {
                    
                    context.setCurrentType(m.getReturnType());
                    context.setCurrentAccessor(OgnlRuntime.getCompiler().getSuperOrInterfaceClass(m, m.getDeclaringClass()));
                    
                    _coreExpression = toSetSourceString(context, target);
                    if (_coreExpression == null || _coreExpression.length() < 1)
                        throw new UnsupportedCompilationException("can't find suitable getter method");
                    
                    _coreExpression +=  ";";
                    _lastExpression = "null";
                    
                    return _coreExpression;
                }
                
                return "";
            } else {
                
                _getterClass = m.getReturnType();
            }

            result = "." + m.getName() + "(";
            
            if ((_children != null) && (_children.length > 0)) {
                
                Class[] parms = m.getParameterTypes();
                String prevCast = (String)context.remove(ExpressionCompiler.PRE_CAST);
/*
                System.out.println("before children methodName is " + _methodName + " for target " + target + " target class: " + (target != null ? target.getClass() : null)
                           + " current type: " + context.getCurrentType() + " and previous type: " + context.getPreviousType());*/

                for(int i = 0; i < _children.length; i++) {
                    if (i > 0) {
                        result = result + ", ";
                    }

                    Class prevType = context.getCurrentType();

                    context.setCurrentObject(context.getRoot());
                    context.setCurrentType(context.getRoot() != null ? context.getRoot().getClass() : null);
                    context.setCurrentAccessor(null);
                    context.setPreviousType(null);

                    Object value = _children[i].getValue(context, context.getRoot());
                    String parmString = _children[i].toGetSourceString(context, context.getRoot());
                    
                    if (parmString == null || parmString.trim().length() < 1)
                        parmString = "null";

                    // to undo type setting of constants when used as method parameters
                    if (ASTConst.class.isInstance(_children[i])) {
                        
                        context.setCurrentType(prevType);
                    }

                    parmString = ExpressionCompiler.getRootExpression(_children[i], context.getRoot(), context) + parmString;

                    String cast = "";
                    if (ExpressionCompiler.shouldCast(_children[i])) {

                        cast = (String)context.remove(ExpressionCompiler.PRE_CAST);
                    }
                    if (cast == null)
                        cast = "";
                    
                    if (!ASTConst.class.isInstance(_children[i]))
                        parmString = cast + parmString;
                    
                    Class valueClass = value != null ? value.getClass() : null;
                    if (NodeType.class.isAssignableFrom(_children[i].getClass()))
                        valueClass = ((NodeType)_children[i]).getGetterClass();

                    if (valueClass != parms[i]) {
                        
                        if (parms[i].isArray()) {

                            parmString = OgnlRuntime.getCompiler().createLocalReference(context,
                                    "(" + ExpressionCompiler.getCastString(parms[i])
                                    + ")ognl.OgnlOps#toArray(" + parmString + ", " + parms[i].getComponentType().getName()
                                    + ".class, true)",
                                    parms[i]
                            );
                            
                        } else  if (parms[i].isPrimitive()) {
                            
                            Class wrapClass = OgnlRuntime.getPrimitiveWrapperClass(parms[i]);
                            
                            parmString = OgnlRuntime.getCompiler().createLocalReference(context,
                                    "((" + wrapClass.getName()
                                    + ")ognl.OgnlOps#convertValue(" + parmString + ","
                                    + wrapClass.getName() + ".class, true))."
                                    + OgnlRuntime.getNumericValueGetter(wrapClass),
                                    parms[i]
                            );

                        } else if (parms[i] != Object.class) {
                            parmString = OgnlRuntime.getCompiler().createLocalReference(context,
                                    "(" + parms[i].getName() + ")ognl.OgnlOps#convertValue(" + parmString + "," + parms[i].getName() + ".class)",
                                    parms[i]
                            );
                        } else if ((NodeType.class.isInstance(_children[i])
                                && ((NodeType)_children[i]).getGetterClass() != null 
                                && Number.class.isAssignableFrom(((NodeType)_children[i]).getGetterClass()))
                                || (valueClass != null && valueClass.isPrimitive()))
                        {
                            parmString = " ($w) " + parmString;
                        } else if (valueClass != null && valueClass.isPrimitive())
                        {
                            parmString = "($w) " + parmString;
                        }
                    }
                    
                    result += parmString;
                }

                if (prevCast != null) {
                    
                    context.put(ExpressionCompiler.PRE_CAST, prevCast);
                }
            }
            
        } catch (Throwable t)
        {
            throw OgnlOps.castToRuntime(t);
        }

        try {

            Object contextObj = getValueBody(context, target);
            context.setCurrentObject(contextObj);
            
        } catch (Throwable t) {
            // ignore 
        }

        result += ")" + post;

        if (m.getReturnType() == void.class) {
            _coreExpression = result + ";";
            _lastExpression = "null";
        }

        context.setCurrentType(m.getReturnType());
        context.setCurrentAccessor(OgnlRuntime.getCompiler().getSuperOrInterfaceClass(m, m.getDeclaringClass()));

        return result;
    }

    public String toSetSourceString(OgnlContext context, Object target)
    {
        Method m = OgnlRuntime.getWriteMethod(context.getCurrentType() != null ? context.getCurrentType() : target.getClass(), _methodName, _children != null ? _children.length : -1);
        if (m == null) {
            return "";
        }

        String post = "";
        String result = "." + m.getName() + "(";
        
        if (m.getReturnType() != void.class && m.getReturnType().isPrimitive()
                && (_parent == null || !ASTTest.class.isInstance(_parent))) {
            
            Class wrapper = OgnlRuntime.getPrimitiveWrapperClass(m.getReturnType());
            
            ExpressionCompiler.addCastString(context, "new " + wrapper.getName() + "(");
            post = ")";
            _getterClass = wrapper;
        } 
        
        try {

            if ((_children != null) && (_children.length > 0)) {

                Class[] parms = m.getParameterTypes();
                String prevCast = (String)context.remove(ExpressionCompiler.PRE_CAST);
                
                for(int i = 0; i < _children.length; i++) {

                    if (i > 0) {
                        result += ", ";
                    }

                    Class prevType = context.getCurrentType();

                    context.setCurrentObject(context.getRoot());
                    context.setCurrentType(context.getRoot() != null ? context.getRoot().getClass() : null);
                    context.setCurrentAccessor(null);
                    context.setPreviousType(null);

                    Object value = _children[i].getValue(context, context.getRoot());
                    String parmString = _children[i].toSetSourceString(context, context.getRoot());

                    if (context.getCurrentType() == Void.TYPE || context.getCurrentType() == void.class)
                        throw new UnsupportedCompilationException("Method argument can't be a void type.");

                    if (parmString == null || parmString.trim().length() < 1) {

                        if (ASTProperty.class.isInstance(_children[i]) || ASTMethod.class.isInstance(_children[i])
                                || ASTStaticMethod.class.isInstance(_children[i]) || ASTChain.class.isInstance(_children[i]))
                            throw new UnsupportedCompilationException("ASTMethod setter child returned null from a sub property expression.");
                        
                        parmString = "null";
                    }

                    // to undo type setting of constants when used as method parameters
                    if (ASTConst.class.isInstance(_children[i])) {

                        context.setCurrentType(prevType);
                    }

                    parmString = ExpressionCompiler.getRootExpression(_children[i], context.getRoot(), context) + parmString;

                    String cast = "";
                    if (ExpressionCompiler.shouldCast(_children[i])) {

                        cast = (String)context.remove(ExpressionCompiler.PRE_CAST);
                    }
                    if (cast == null)
                        cast = "";
                    
                    parmString = cast + parmString;

                    Class valueClass = value != null ? value.getClass() : null;
                    if (NodeType.class.isAssignableFrom(_children[i].getClass()))
                        valueClass = ((NodeType)_children[i]).getGetterClass();
                    
                    if (valueClass != parms[i]) {
                        
                        if (parms[i].isArray()) {

                            parmString = OgnlRuntime.getCompiler().createLocalReference(context,
                                    "(" + ExpressionCompiler.getCastString(parms[i])
                                    + ")ognl.OgnlOps#toArray(" + parmString + ", " + parms[i].getComponentType().getName()
                                    + ".class)",
                                    parms[i]
                            );

                        } else  if (parms[i].isPrimitive()) {
                            
                            Class wrapClass = OgnlRuntime.getPrimitiveWrapperClass(parms[i]);
                            
                            parmString = OgnlRuntime.getCompiler().createLocalReference(context,
                                    "((" + wrapClass.getName()
                                    + ")ognl.OgnlOps#convertValue(" + parmString + ","
                                    + wrapClass.getName() + ".class, true))."
                                    + OgnlRuntime.getNumericValueGetter(wrapClass),
                                    parms[i]
                            );
                            
                        } else if (parms[i] != Object.class) {

                            parmString = OgnlRuntime.getCompiler().createLocalReference(context,
                                    "(" + parms[i].getName() + ")ognl.OgnlOps#convertValue(" + parmString + "," + parms[i].getName() + ".class)",
                                    parms[i]
                            );
                            
                        } else if ((NodeType.class.isInstance(_children[i])
                                && ((NodeType)_children[i]).getGetterClass() != null
                                && Number.class.isAssignableFrom(((NodeType)_children[i]).getGetterClass()))
                                || (valueClass != null && valueClass.isPrimitive()))
                        {
                            parmString = " ($w) " + parmString;
                        } else if (valueClass != null && valueClass.isPrimitive()) 
                        {
                            parmString = "($w) " + parmString;
                        }
                    }
                    
                    result += parmString;
                }

                if (prevCast != null)
                {
                    context.put(ExpressionCompiler.PRE_CAST, prevCast);
                }
            }
            
        } catch (Throwable t)
        {
            throw OgnlOps.castToRuntime(t);
        }

        try {

            Object contextObj = getValueBody(context, target);
            context.setCurrentObject(contextObj);

        } catch (Throwable t) {
            // ignore
        }

        context.setCurrentType(m.getReturnType());
        context.setCurrentAccessor(OgnlRuntime.getCompiler().getSuperOrInterfaceClass(m, m.getDeclaringClass()));
        
        return result + ")" + post;
    }
}
